package cn.tedu.boot02.controller;

import cn.tedu.boot02.DBUtil.DBUtil;
import cn.tedu.boot02.entity.Product;
import org.springframework.web.bind.annotation.RequestMapping;
import org.springframework.web.bind.annotation.RestController;

import java.sql.Connection;
import java.sql.PreparedStatement;
import java.sql.ResultSet;
import java.sql.SQLException;
import java.util.ArrayList;

@RestController
public class ProductController {

    @RequestMapping("/insert")
    public String insert(Product product) {
        System.out.println("product = " + product);

        try (
                Connection conn = DBUtil.getConnection()
        ) {
            String sql = "INSERT INTO product VALUES (NULL,?,?,?)";
            PreparedStatement statement = conn.prepareStatement(sql);
            statement.setString(1, product.getTitle());
            statement.setDouble(2, product.getPrice());
            statement.setInt(3, product.getNum());

            statement.executeUpdate();
        } catch (SQLException throwables) {
            throwables.printStackTrace();
        }
        return "添加完成";
    }

    @RequestMapping("/select")
    public String select() {
        ArrayList<Product> list = new ArrayList<>();
        try (
                Connection conn = DBUtil.getConnection();
        ) {
            String sql = "SELECT * FROM product;";
            PreparedStatement ps = conn.prepareStatement(sql);
            ResultSet rs = ps.executeQuery();
            while (rs.next()) {
                Product product = new Product();
                product.setId(rs.getInt(1));
                product.setTitle(rs.getString(2));
                product.setPrice(rs.getDouble(3));
                product.setNum(rs.getInt(4));
                list.add(product);
            }
        } catch (SQLException throwables) {
            throwables.printStackTrace();
        }
        String html = "<table border='1'> " +
                "   <caption>商品列表</caption> " +
                "  <tr> " +
                "    <th>id</th> " +
                "    <th>标题</th> " +
                "    <th>价格</th> " +
                "    <th>库存</th> " +
                "    <th>操作</th> " +
                "  </tr> ";
        for (Product p : list) {
            html += "   <tr> " +
                    "   <td>" + p.getId() + "</td> " +
                    "   <td>" + p.getTitle() + "</td> " +
                    "   <td>" + p.getPrice() + "</td> " +
                    "   <td>" + p.getNum() + "</td> " +
                    "   <td><a href='/delete?id=" + p.getId() + "'>删除</a></td> " +
                    "  </tr>";
        }
        html += "</table>";
        return html;
    }

    @RequestMapping("/delete")
    public String delete(int id) {
        System.out.println("id = " + id);

        try (
                Connection connection = DBUtil.getConnection();
        ) {
            String sql = "DELETE FROM product WHERE id = ?";
            PreparedStatement ps = connection.prepareStatement(sql);
            ps.setInt(1, id);
            ps.executeUpdate();
        } catch (SQLException throwables) {
            throwables.printStackTrace();
        }

        return "删除成功！<a href='/select'>返回列表</a>";
    }

    @RequestMapping("/update")
    public String update(Product product) {
        try (Connection conn = DBUtil.getConnection()) {
            String sql = "UPDATE product SET title=?,price=?,num=? WHERE id=?";
            PreparedStatement ps = conn.prepareStatement(sql);
            ps.setString(1, product.getTitle());
            ps.setDouble(2, product.getPrice());
            ps.setInt(3, product.getNum());
            ps.setInt(4, product.getId());
            ps.executeUpdate();
        } catch (SQLException throwables) {
            throwables.printStackTrace();
        }
        return "修改完成！<a href='/select'>返回列表页面</a>";
    }
}
